Method: Torch::Tensor#[]

Defined in:
lib/torch/tensor.rb

#[](*indexes) ⇒ Object

based on python_variable_indexing.cpp and pytorch.org/cppdocs/notes/tensor_indexing.html


167
168
169
170
 
# File 'lib/torch/tensor.rb', line 167

def [](*indexes)
  indexes = indexes.map { |v| v.is_a?(Array) ? Torch.tensor(v) : v }
  _index(indexes)
end